from setuptools import setup
from torch.utils import cpp_extension
import os

cpp_src = ['my_conv.cpp', 'my_conv_hip.hip']

if __name__ == '__main__':

    setup(
        name='myconv',
        ext_modules=[
            cpp_extension.CUDAExtension(
                'myconv', cpp_src,
                extra_compile_args={
                 "cxx":["-std=c++17"],
                 "nvcc":["-std=c++17"]
                })
        ],
        cmdclass={'build_ext': cpp_extension.BuildExtension})
